Fix type of check