Slight API update