Code fix

- Fixed a line that was incomplete