Fixed bug where if statment did not return anything

Signed-off-by: epicfinley <epicfinley@gmail.com>