Why not just do an if else for this?