Small fix to controller