Fixed bug; used missing module