Quick fix, need to make something more efficient later.