Fix controller

Copied into wrong file..