Small fix to test control