Fixed minor js issue