Change this to be an `if` statement