Fixed code showing bug