Minor Fix

- Changed version check