woops... minor bug fix