Fixed code not blocking properly.