Fixed checks to pass test

Signed-off-by: Joseph Lee <josel@pdx.edu>