- fixed code bug