Fix

Fix wrong check for the prime numbers