Fixed bug in source code