Fixed prototype issue