Fixed code due to githug bug