use an if/else for these two lines.