Fixed style code bug