Fixed more things

- Added one statement