diff --git a/scheduler.go b/scheduler.go
index bd7c943..20ce644 100644
--- a/scheduler.go
+++ b/scheduler.go
@@ -56,7 +56,7 @@ func main() {
 		fmt.Println(task)
 	}
 
-	scheduler := schedulers.NewPistonCapper(tasks, *ignoreWatts)
+	scheduler := schedulers.NewBPSWClassMapWatts(tasks, *ignoreWatts)
 	driver, err := sched.NewMesosSchedulerDriver(sched.DriverConfig{
 		Master: *master,
 		Framework: &mesos.FrameworkInfo{